package com.zq.solution;

import java.util.*;

/**
 * @ClassName LadderLength
 * @Description
 * 127. 单词接龙
 *  给定两个单词（beginWord 和 endWord）和一个字典，找到从 beginWord 到 endWord 的最短转换序列的长度。转换需遵循如下规则：
 *      1、每次转换只能改变一个字母。
 *      2、转换过程中的中间单词必须是字典中的单词。
 *  说明:
 *       1.如果不存在这样的转换序列，返回 0。
 *       2.所有单词具有相同的长度。
 *       3.所有单词只由小写字母组成。
 *       4.字典中不存在重复的单词。
 *       5.你可以假设 beginWord 和 endWord 是非空的，且二者不相同。
 *  示例:
 *       输入:
 *          beginWord = "hit",
 *          endWord = "cog",
 *          wordList = ["hot","dot","dog","lot","log","cog"]
 *       输出: 5
 *       解释: 一个最短转换序列是 "hit" -> "hot" -> "dot" -> "dog" -> "cog", 返回它的长度 5。
 * @Author ZQ
 * @Date 2020/11/5 14:36
 * @Version V1.0
 **/
public class LadderLength {
    static Map<String, Integer> wordId = new HashMap<String, Integer>();
    static List<List<Integer>> edge = new ArrayList<List<Integer>>();
    static int nodeNum = 0;

    public static int ladderLength(String beginWord, String endWord, List<String> wordList) {
        for (String word : wordList) {
            addEdge(word);
        }
        addEdge(beginWord);
        if (!wordId.containsKey(endWord)) {
            return 0;
        }
        int[] dis = new int[nodeNum];
        Arrays.fill(dis, Integer.MAX_VALUE);
        int beginId = wordId.get(beginWord), endId = wordId.get(endWord);
        dis[beginId] = 0;

        Queue<Integer> que = new LinkedList<Integer>();
        que.offer(beginId);
        while (!que.isEmpty()) {
            int x = que.poll();
            if (x == endId) {
                return dis[endId] / 2 + 1;
            }
            for (int it : edge.get(x)) {
                if (dis[it] == Integer.MAX_VALUE) {
                    dis[it] = dis[x] + 1;
                    que.offer(it);
                }
            }
        }
        return 0;
    }

    public static void addEdge(String word) {
        addWord(word);
        int id1 = wordId.get(word);
        char[] array = word.toCharArray();
        int length = array.length;
        for (int i = 0; i < length; ++i) {
            char tmp = array[i];
            array[i] = '*';
            String newWord = new String(array);
            addWord(newWord);
            int id2 = wordId.get(newWord);
            edge.get(id1).add(id2);
            edge.get(id2).add(id1);
            array[i] = tmp;
        }
    }

    public static void addWord(String word) {
        if (!wordId.containsKey(word)) {
            wordId.put(word, nodeNum++);
            edge.add(new ArrayList<Integer>());
        }
    }

    public static void main(String[] args) {
        List<String> list = Arrays.asList("hot","dot","dog","lot","log","cog");
        System.out.println(ladderLength("hit","cog",list));
    }
}
